import re
from collections import defaultdict

# 题目描述：对输入一段英文文章，单词之间用空格或标点符号（string.punctuation）隔开，Hello, world!Welcome to Python
str_content='Hello,world world world!Welcome Welcome to Python'
arr=re.findall(r'[a-zA-z]+',str_content)
# print(arr)
# dict_count=defaultdict(int)
dict_count=dict()
for e in arr:
    dict_count.setdefault(e,0)
    dict_count[e]+=1
# print(dict_count)
arr_count=list(dict_count.items())
n=2
# print(dict_count.items())
arr_count.sort(key=lambda a:a[1],reverse=True)
# print(arr_count)
# 统计该段文章中的每个单词出现的次数（以小写为准），并在当前目录中创建一个文件"Wordtj.”，
# 将出
# 现次数最多的n个单词及其出现次数，以“单词，次数”（使用英文逗号分隔）格式分行存入Wordtj.
# 文件中。
# 输入格式：在第一行中输入一个正整数n（1n10）；在第二行中输入一个段英文文章。
# 输出格式：文件。

with open('Wordtj.csv', 'w') as f:

    str_title = 'word,count\n'
    f.writelines(str_title)
    for e in arr_count[0:n]:
        # str_line = e[0] + "," + str(e[1]) + '\n'
        str_line = f"{e[0]},{e[1]}\n"
        f.writelines(str_line)
